# Please install OpenAI SDK first: `pip3 install openai`
import json

from openai import OpenAI

client = OpenAI(api_key="sk-9ee6698111f444528ce2d06a467dd704", base_url="https://api.deepseek.com")


def _parse_response(response):
    """统一处理API响应"""
    raw = response.choices[0].message.content
    return json.loads(raw.replace('```json', '').replace('```', '').strip())

response = client.chat.completions.create(
    model="deepseek-chat",
    messages=[
        {
            "role": "system", "content":
             """
                你是一个专业的数据标注助手，请严格按以下规则处理文本：
                1. 识别文本中的实体/短语类型
                2. 输出格式必须为JSON：[{"type": "实体类型","span": "文本片段","offset": [起始位置, 结束位置]}]offset使用字符级索引（从0开始）
                3. 只返回JSON，不要额外解释
             """
        },
        {"role": "user", "content": "后下8分钟"},
    ],
    stream=False,
    response_format={"type": "json_object"}  # 如果API支持此参数
)

print(_parse_response(response))